area of square in python